import unittest
import sys,  os
from pylab import *
sys.path.append( os.getcwd() )
from archivocsv import *


class TestArchivoCSV(unittest.TestCase):
  def setUp(self):    
    # Custodio
    #r = 10
    #t = [1|3|6|10|20|40|70|100|200|400|700|1000|2000|3000]
    #ti = []
    #s = [3.4,4.2,4.8,5.08,5.60,6.05,6.40,6.65,7.10,7.50,7.90,8.25,8.70,8.90]
    #Q = 100.0/1000.0*60*60*24.0
    self.data = '1|3.4\n3|4.2\n'
    self.sep = r'|'
    self.tienehead = True

  def testInit(self):
    d = ArchivoCSV(self.data, self.sep, self.tienehead)
    t = d.getTiempo()
    s = d.getDescenso()
    self.assertEqual(t[0], '1')
    self.assertEqual(t[1], '3')
    self.assertEqual(s[0], '3.4')
    self.assertEqual(s[1], '4.2')
    
if __name__ == '__main__':
    unittest.main()
